'''
Created on 12/10/2011

@author: silviojr
'''
from Thesaurus import *
from dicionario import *
import cPickle as pickle

if __name__ == '__main__':
    pass

thesaurus = Thesaurus()
'''

f = open("/home/silviojr/poc2/base_tep2/triplos.txt", "r")
lines = f.readlines()

for line in lines:
    linha = line.split(" ")
    linha[3] = linha[3][:-2]
    thesaurus.insereTripla(linha[0], linha[1], linha[2], linha[3])
    thesaurus.insereTripla(linha[0], linha[3], linha[2], linha[1])
    #print linha[0], " ", linha[1], " ", linha[2], " ", linha[3]


pickle.dump(thesaurus, open("/home/silviojr/poc2/thesaurus2.p", "wb"))
''' 
thesaurusCarregado = pickle.load(open("/home/silviojr/poc2/thesaurus2.p", "rb"))

#thesaurusCarregado.imprimeThesaurus()


listaSeed = []


termo1 =  Termo("bom", "[Adjetivo]", "positiva", 0, "seed", "seed", "seed")
termo2 =  Termo("bonito", "[Adjetivo]", "positiva", 0, "seed", "seed", "seed")

listaSeed.append(termo1)
listaSeed.append(termo2)

listaSeed.append(Termo("gostar", "[Verbo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("amar", "[Verbo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("bondade", "[Substantivo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("superior", "[Adjetivo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("superioridade", "[Substantivo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("excelente", "[Adjetivo]", "positiva", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("legal", "[Adjetivo]", "positiva", 0, "seed", "seed", "seed"))

listaSeed.append(Termo("ruim", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("mau", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("feio", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("odiar", "[Verbo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("maldade", "[Substantivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("negativo", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("errado", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("inferior", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("sujo", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("arrogante", "[Adjetivo]", "negativa", 0, "seed", "seed", "seed"))
listaSeed.append(Termo("esnobismo", "[Substantivo]", "negativa", 0, "seed", "seed", "seed"))

for i in range(1,10):
    print '**************************nivel ',i ,'**************************************'
    dicionario = Dicionario(i)
    dicionario.expandeDicionario(thesaurusCarregado, listaSeed)
    arquivo = "/home/silviojr/poc2/dicionario"+ str(i) + ".p"
    pickle.dump(dicionario, open(arquivo, "wb"))

    #thesaurusCarregado.procuraTermo("nocivo")

    dicionarioCarregado = pickle.load(open(arquivo, "rb"))
    dicionarioCarregado.imprimeDicionario("votos", listaSeed)
    print '---------------------------------------------------'
    dicionarioCarregado.imprimeDicionario("nivel", listaSeed)
